tensor([[0.3900, 0.3900, 0.3900, 0.3900, 0.3900],
        [0.6100, 0.6100, 0.6100, 0.6100, 0.6100]], grad_fn=<RoundBackward1>)